"""
学员管理系统
"""

users = [{"name":"xiaoluo","age":22,"address":"SH"}]
def printInfo():
    print("-"*50)
    print("欢迎登录学员管理系统".center(50))
    print("1：添加学员")
    print("2：删除学员")
    print("3：修改学员信息")
    print("4：查询学员信息")
    print("5：显示所有学员的信息")
    print("6：退出系统")
    print("-"*50)
def addUser():
    while True:
        isAdd = input("请输入您的选择：1 添加学员 ，2 退出添加")
        if(isAdd == '1'):
            print("添加学员：")
            name = input("请输入学员姓名：")
            age=input("请输入学员年龄：")
            address=input("请输入学员的家庭地址：")
            user = {
                "name" : name,
                "age":age,
                "address":address
            }
            users.append(user)
            print(f"学员添加成功：{user}")
        elif (isAdd == '2'):
            break
        else:
            print("添加学员->无效操作直接退出")
            break

def delUserByName():
    while True:
        isAdd = input("请输入您的选择：1 删除学员 ，2 退出删除")
        if(isAdd == '1'):
            print("删除学员：")
            name = input("请输入学员姓名：")
            for i in range(len(users)):
                if users[i]["name"] == name :
                    del users[i]
                    break
            print(f"{name}信息删除成功！")
        elif (isAdd == '2'):
            break
        else:
            print("添加学员->无效操作直接退出")
            break

def updateUserByName():
    while True:
        isUpdate = input("请输入您的选择：1 修改学员 ，2 退出修改")
        if (isUpdate == '1'):
            user = None
            userIndex = -1
            username = input("请输入学员姓名：")
            for i in range(len(users)):
                if users[i]['name'] == username:
                    user = users[i]
                    userIndex =  i
                    print(f"学员信息{user}")
                    break
            if user == None:
                print("未找到该学员")
                return
            # 如果找到这个学员
            print("* * "*4 + "开始修改" + "* * "*4)
            user['name'] = input("请输入新的姓名：")
            user['age'] = input("请输入新的年龄：")
            user['address'] = input("请输入新的地址：")
            users[userIndex] = user
        elif (isUpdate == '2'):
            break
        else:
            print("添加学员->无效操作直接退出")
            break


def getInfoByName():
    while True:
        isQuery = input("请输入您的选择：1 查询用户 ，2 退出查询")
        if (isQuery == '1'):
            user = None
            username = input("请输入学员姓名：")
            for i in range(len(users)):
                if users[i]['name'] == username:
                    user = users[i]
                    print(f"学员信息{user}")
                    break
            if user == None:
                print("！！！！！！未找到该学员！！！！！！")
                return
        elif (isQuery == '2'):
            break
        else:
            print("添加学员->无效操作直接退出")
            break


while True:
    printInfo()
    choice = input("请输入您的选择：")
    if(choice == '1' ):
        addUser()
    elif(choice == '2' ):
        delUserByName()
    elif (choice == '3'):
        updateUserByName()
    elif (choice == '4'):
        getInfoByName()
    elif (choice == '5'):
        print(users)
    elif (choice == '6'):
        isExit = input("确认退出吗？ enter yes to exit ")
        if isExit == 'yes':
            break
    else:
        print("无效操作~")

